package config

import "github.com/spf13/viper"

var SECRET_KEY string           //jwt加密密钥
var SECRET_REFRESH_KEY string   //jwt刷新令牌加密密钥
var OPENAI_API_KEY string       //open
var BASE_PROMPT_TEMPLATE string //管理员评论时，给AI的基本提示词模板

var RECOMMENDED_MOVIE_LIMIT int64 //为用户推荐电影时，默认最多推荐的数量

func initEnv() {
	viper.SetConfigFile("./config/.env")
	viper.SetConfigType("env") //设置解析类型为环境变量

	//读取配置文件
	if err := viper.ReadInConfig(); err != nil {
		panic("环境变量的配置文件读取失败,err: " + err.Error())
	}

	SECRET_KEY = viper.GetString("SECRET_KEY")
	SECRET_REFRESH_KEY = viper.GetString("SECRET_REFRESH_KEY")
	OPENAI_API_KEY = viper.GetString("OPENAI_API_KEY")
	BASE_PROMPT_TEMPLATE = viper.GetString("BASE_PROMPT_TEMPLATE")

	RECOMMENDED_MOVIE_LIMIT = viper.GetInt64("RECOMMENDED_MOVIE_LIMIT")
}
